//给你一个整数数组 nums 和两个整数 k 和 t 。请你判断是否存在 两个不同下标 i 和 j，使得 abs(nums[i] - nums[j]) <= 
//t ，同时又满足 abs(i - j) <= k 。 
//
// 如果存在则返回 true，不存在返回 false。 
//
// 
//
// 示例 1： 
//
// 
//输入：nums = [1,2,3,1], k = 3, t = 0
//输出：true 
//
// 示例 2： 
//
// 
//输入：nums = [1,0,1,1], k = 1, t = 2
//输出：true 
//
// 示例 3： 
//
// 
//输入：nums = [1,5,9,1,5,9], k = 2, t = 3
//输出：false 
//
// 
//
// 提示： 
//
// 
// 0 <= nums.length <= 2 * 104 
// -231 <= nums[i] <= 231 - 1 
// 0 <= k <= 104 
// 0 <= t <= 231 - 1 
// 
// Related Topics 数组 桶排序 有序集合 排序 滑动窗口 
// 👍 469 👎 0

/**
 * @author DaHuangXiao
 */
package leetcode.editor.cn;

import java.util.HashSet;
import java.util.Set;
import java.util.TreeSet;

public class ContainsDuplicateIii {
    public static void main(String[] args) {
        Solution solution = new ContainsDuplicateIii().new Solution();
    }

    //leetcode submit region begin(Prohibit modification and deletion)

    class Solution {
        // 先弄清楚题目的意思
        // 用红黑树加快搜索
        // N(nlogk)
        public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
            int left = 0;
            int right = 0;
            int n = nums.length;
            TreeSet<Long> set = new TreeSet<>();
            while (right<n){
                while (right-left>k){
                    set.remove(nums[left]*1L);
                    left++;
                }
                if (isSafity(set,nums[right]*1L,t)){
                    return true;
                }
                set.add(nums[right]*1L);
                right++;
            }
            return false;
        }

        private boolean isSafity(TreeSet<Long> treeSet, long cur,int t) {
            Long ceiling = treeSet.ceiling(cur);
            Long floor = treeSet.floor(cur);
            if (ceiling != null && ceiling - cur <= t) {
                return true;
            }
            if (floor != null && cur - floor <= t) {
                return true;
            }
            return false;
        }

    }
//leetcode submit region end(Prohibit modification and deletion)

}